from selenium import webdriver
from time import sleep
from selenium.webdriver.common.by import By

class LikeShopTest(): 
#     def init(self):
#         self.browser = webdriver.Firefox()

    def test(self):
        self.browser = webdriver.Firefox()
        self.browser.get("http://likeshop.abc/admin/account/login.html")
        sleep(3)
        self.browser.find_element(By.NAME, 'account').send_keys('tgc')
        self.browser.find_element(By.NAME, 'password').send_keys('admin')
        self.browser.find_element(By.NAME, 'code').send_keys('1111')
        self.browser.find_element(By.ID, 'login').click()
        sleep(3)
        self.browser.find_element(By.LINK_TEXT, '商品').click()

        framel = self.browser.find_element(By.XPATH, '//*[@id="LAY_app_body"]/div[2]/iframe')
        self.browser.switch_to.frame(framel)
        sleep(1)
        self.browser.find_element(By.XPATH, '//*[@id="keyword"]').send_keys('文具')
        sleep(1)
        self.browser.find_element(By.XPATH, '/html/body/div/div/div[2]/div/div/div[1]/div/div[5]/button[1]').click()
        sleep(2)
        shop_list = self.browser.find_elements(By.XPATH, '/html/body/div/div/div[2]/div/div/div[2]/div[2]/div[1]/div[2]/table/tbody/tr')
        count = len(shop_list)
        if count == 0:
            print('没有数据')
        else:
            shop_name_ret = True
            for shop_item in shop_list:
                shop_name = shop_item.find_element(By.XPATH, './td[2]/div').text
                if '文具' not in shop_name:
                        shop_name_ret = False
                for i in range(1, count + 1):
                    xpath = '/html/body/div/div/div[2]/div/div/div[2]/div[2]/div[1]/div[2]/table/tbody/tr[%s]/td[2]/div' %i
                    shop_name = self.browser.find_element(By.XPATH, xpath).text
                    if '文具' not in shop_name:
                        shop_name_ret = False
                if shop_name_ret:
                    print('没有问题')
                else:
                    print('有bug')
            sleep(3)

likeshop = LikeShopTest()
likeshop.test()